FormatTranspose

将输入数据从一种存储格式转换为另一种存储格式，如 NCHW ↔ NHWC、NC4HW4、NC8HW8 等，适用于图像或特征图数据。

输入：

  • src_data - 输入数据地址。

  • src_format - 输入数据格式标识。

  • dst_format - 输出数据格式标识。

  • batch - 批大小。

  • channel - 通道数。

  • plane - 高*宽。

  • core_mask - 核掩码（仅适用于共享存储版本）。

输出：

  • dst_data - 格式转换结果地址。

支持平台：

FT78NE MT7004

备注

  • FT78NE 支持fp, dp, int8, int16, int32, clx64, cplx128

  • MT7004 支持hp, fp, i16, i32, cplx64

共享存储版本:

void fp_formattranspose_s(int src_format, int dst_format, float *src_data, float *dst_data, int batch, int channel, int plane, int core_mask)
void hp_formattranspose_s(int src_format, int dst_format, half *src_data, half *dst_data, int batch, int channel, int plane, int core_mask)
void dp_formattranspose_s(int src_format, int dst_format, double *src_data, double *dst_data, int batch, int channel, int plane, int core_mask)
void i8_formattranspose_s(int src_format, int dst_format, int8_t *src_data, int8_t *dst_data, int batch, int channel, int plane, int core_mask)
void i16_formattranspose_s(int src_format, int dst_format, int16_t *src_data, int16_t *dst_data, int batch, int channel, int plane, int core_mask)
void i32_formattranspose_s(int src_format, int dst_format, int *src_data, int *dst_data, int batch, int channel, int plane, int core_mask)
void c64_formattranspose_s(int src_format, int dst_format, float *src_data, float *dst_data, int batch, int channel, int plane, int core_mask)
void c128_formattranspose_s(int src_format, int dst_format, double *src_data, double *dst_data, int batch, int channel, int plane, int core_mask)

C调用示例：

 1#include <stdio.h>
 2#include <formattranspose.h>
 3
 4int main() {
 5    float *input = (float *)0xA0000000;      // 输入在DDR空间
 6    float *output = (float *)0xC0000000;
 7    int batch = 1, channel = 3, plane = 224*224;
 8    int src_format = 1; // NHWC
 9    int dst_format = 0; // NCHW
10    int core_mask = 0xff;
11
12    fp_formattranspose_s(src_format, dst_format, input, output, batch, channel, plane, core_mask);
13    return 0;
14}

私有存储版本:

void fp_formattranspose_p(int src_format, int dst_format, float *src_data, float *dst_data, int batch, int channel, int plane)
void hp_formattranspose_p(int src_format, int dst_format, half *src_data, half *dst_data, int batch, int channel, int plane)
void dp_formattranspose_p(int src_format, int dst_format, double *src_data, double *dst_data, int batch, int channel, int plane)
void i8_formattranspose_p(int src_format, int dst_format, int8_t *src_data, int8_t *dst_data, int batch, int channel, int plane)
void i16_formattranspose_p(int src_format, int dst_format, int16_t *src_data, int16_t *dst_data, int batch, int channel, int plane)
void i32_formattranspose_p(int src_format, int dst_format, int *src_data, int *dst_data, int batch, int channel, int plane)
void c64_formattranspose_p(int src_format, int dst_format, float *src_data, float *dst_data, int batch, int channel, int plane)
void c128_formattranspose_p(int src_format, int dst_format, double *src_data, double *dst_data, int batch, int channel, int plane)

C调用示例：

 1#include <stdio.h>
 2#include <formattranspose.h>
 3
 4int main() {
 5    float *input = (float *)0x10810000;      // 输入在L2空间
 6    float *output = (float *)0x10820000;
 7    int batch = 1, channel = 3, plane = 224*224;
 8    int src_format = 1; // NHWC
 9    int dst_format = 0; // NCHW
10
11    fp_formattranspose_p(src_format, dst_format, input, output, batch, channel, plane);
12    return 0;
13}